hackerrank shape and reshape problem solution in python programming